package com.knowledge.business.model;

import lombok.Data;

import java.util.List;

/**
 * 向量模型响应实体
 *
 * @author knowledge
 */
@Data
public class EmbeddingResponse {

    private String object;
    private List<EmbeddingData> data;
    private String model;
    private Usage usage;


    public static class EmbeddingData {
        private String object;
        private List<Float> embedding;
        private int index;

        public String getObject() {
            return object;
        }

        public void setObject(String object) {
            this.object = object;
        }

        public List<Float> getEmbedding() {
            return embedding;
        }

        public void setEmbedding(List<Float> embedding) {
            this.embedding = embedding;
        }

        public int getIndex() {
            return index;
        }

        public void setIndex(int index) {
            this.index = index;
        }
    }


    @Data
    public static class Usage {
        private Integer prompt_tokens;
        private Integer total_tokens;
        private Integer completion_tokens;


    }
}